package work.base.data.queue;

import java.util.concurrent.LinkedTransferQueue;
import java.util.concurrent.TransferQueue;

/**
 * LinkedTransferQueue的使用示例
 */
public class ProducerConsumerExample {
    public static void main(String[] args) {
        TransferQueue<String> queue = new LinkedTransferQueue<>();

        // 生产者线程
        Thread producer = new Thread(() -> {
            for (int i = 0; i < 3; i++) {
                try {
                    String message = "message" + i;
                    System.out.println("生产者准备发送: " + message);

                    // 使用 tryTransfer() 方法尝试立即传输数据
                    if (queue.tryTransfer(message)) {
                        System.out.println("生产者立即发送成功: " + message);
                    } else {
                        // 如果没有等待的消费者，则使用 transfer() 阻塞等待
                        queue.transfer(message);
                        System.out.println("生产者已发送: " + message);
                    }

                    Thread.sleep(1000);
                } catch (InterruptedException e) {
                    Thread.currentThread().interrupt();
                }
            }
        });

        // 消费者线程
        Thread consumer = new Thread(() -> {
            for (int i = 0; i < 3; i++) {
                try {
                    System.out.println("消费者等待接收数据...");
                    // take() 方法会阻塞直到有数据可用
                    String message = queue.take();
                    System.out.println("消费者接收到: " + message);

                    // 模拟处理数据的时间
                    Thread.sleep(1500);
                } catch (InterruptedException e) {
                    Thread.currentThread().interrupt();
                }
            }
        });

        // 启动线程
        consumer.start();
        producer.start();

        // 等待线程完成
        try {
            producer.join();
            consumer.join();
        } catch (InterruptedException e) {
            Thread.currentThread().interrupt();
        }
    }
}
